Some addiotional edits to assignment 1