Added README
Tidied up code