Made SetModels() virtual