cleaning up code
cleaned up code and updated readme.md