clean up code and edit readme