working submission for HW2.1