Add README
Tidied up code