Cleaning up code and editing readme.