Added additional checks for element classes in wireframe.