HW 2 check corrections