add calculate.py