Typo
Refactored to make use of argparse