Updates

Updates to the script!