JAVA Week 5 solution