Minor fix in HW3. Update HW2 solution