Added in the script and fixed readme