Add edits for Assignment 2