little fixes to w1d4 solutions