added main model class