updated assignment for week 2