Add getDepth function