Updated the readme
Small tidy of the demo code