Fix a bug

Fixed bug with check on number of cmd line args and in argparse input variable retrieval. Copy paste bug I guess.