Week 2 solution