add class differentiation