Added HW3 code