Add calculator.py