Make API work for all model types
Add method override