Add tests for models and attributes