cleaning code and updating Readme