Updated UI add model logic