Minor fixes to Q2 in HW 3