Final edits to assignment 2