First working version of the script