Update 6.get_depth.py