Minor updated to script