add model and fix controller to use it