Cleaning up the script

looks more friendly