Update README.md
Code tidy up