clean up the code
README.md edited