Tidying

Tidied up the repo and some code.
File update