Cleaning up code. Updating readme.