Fixed something in hw2q6