Update main.py

bug fixes