edited Readme and cleaned up code