Added Model for tests