commenting and edits to week 5 code