Submissions for team Neural Knights on problem nah_main