Now for a few changes to the script so it works like I want it to ;)