fixing assignment 2 week 3