HW2_1_b complete