Tidy the code up a little
Tidy
Update readme