add class checking