Updated week 4 assignment