Adding models to represent the data