First stab at HW2
finishing HW2