Update get_depth.py