updating assginment 4