minor chnage to script