worked on HW assignment 1