Update problem1.java