Add search to model
Add search to ui