Update assignment_week_2.R