clean up code, update readme