Fixed the script REALLY happy with how works