Edits for assignment 4