Update script.

- update README.md to reflect the changes
- minor script cleanup