working version of the script