JAVA Week 9 solution