HW4 assignment 1