clean up code, update README