fixed main.py
fixed main.py