Create get_depth.py