updated to match script