there should be a function to compute omega, and it should be passed as a parameter when needed rather than computed in the function polymul_ntt.