Added some stuff to the script