fix week1 assignment