Assignment2 minor edits