Use choices in argparser and follow DRY.
Merge pull request #10 from Raydiation/choices_and_DRY

Use choices in argparser and follow DRY.