Add find and walk methods to tree controller