updated the script
Updated script