Work on part 1 of assignment 1