Updating readme.

Fixing code and cleaning up a little.