Update depth.py