Add tests for TagClasses