MAD 1 Week 5 solution