clean up code and update readme