Submissions for team Fast X Fourier on problem nah_main