MAD 1 Week 4 solution