Update main.py

Basic Calc works