Update main.py

Added argparse support for passing in file as argument.