little update in the script