Added argparse and...

* added argparse so we can feed in info via CLI
* tweaked some comments and added some whitespace