Adding more models
Adding tests for models and a migration