Slight adjustment to script.