hw6 p4 more edits