my edits to assignment 2