Update main.py

Modified by Jane.