Updated README.md and cleaned up code.