Use protocol instead of class, so you can use existing model objects