Updated Week3 assignment