HW 3 assignment