Update

Cleaning up script and removing unnecessary lines.