Update algo.py