Clean up the script and review README