HW4 edits