Made modification to script