Some of the code for assignment 3 part one is done