Updated script to work a bit more nicely.