Main.py now accepts command line arguments