added a bit more to the script