Modified for an argument driven script

Modified for an argument driven script rather than a prompted script.
Also soon to add ability to add a list of hosts rather than a single host at a time.