setting up the model classes