JAVA Week 6 solution