Added level information to the tree controller