Done for HW 1 with no extra credit