added max_depth.py