Minor updates to week 3 assignment 2