Submissions for team Neural Knights on problem lecture_main_main