Updated readme and made a few tweaks to the script