Submissions for team ZWZ on problem nah_main