week1 assigment problem