edited Readme and clean up code