Fix script and update README