Fixed some code and worked on the write up