Use argparse to make version simpler.

Hey, this change simplifies the version argument a little by using functionality built into argparse. PTAL when you get a chance.