Added 2 models for tests