JAVA Week 1 solution