Added new models for the application