set models for each class