update to week4 assignment