JAVA Week 3 solution