JAVA Week 4 solution