Update test.py
Add algo