update week3 assignment