Submissions for team Neural Knots on problem nah_main