README update main.py update