added checks for classes in the wireframe