add class in code