MAD 1 Week 3 solution