I think having the classes is fine they can just be marker classes?