Week 2 task 1 solution