Week 2 solution - simple