Added argparse to main.py.