Tidy up of code and added readme