Updating README.md and cleaning up code a little bit.