Update main.py

Calc Screen