Altered model_classes method