add a class check