Assignment 1- update 2