add class Logic