Use constructor to reference the model class