Adding test model classes