update test for applying classes to treemap