remove main from the model class