add class for tree