Added the models for the application