Submissions for team JAX on problem files_main