Modified the script