Update surf2.py