Edits for assignment 2