Parse arguments using argparse. Add docstrings.