Add compute.py