edits to assingment 4