JAVA Week 2 solution