Instead of using classes, use an enum