Updated tree methods to take IObject instead of IClass