Tidy up code
Add readme