HW2 assignment