Still trying to fix script