updated compute.py