Add generator for subclasses