Submissions for team Neural Knights on problem cross_main