Correct Version of Iteration 1 code

Had to upload iteration 1 code again as some files were overwritten when
I uploaded iteration 2 code