updated README to reflect main.py change