Update README.md. Clean up code.