Updated readme
clean up code