Update main.py

Small Progress...