add checkClass logic