Updated main.py to use supplied input file