main.py updated for command line arguments.