Update calculations.py