Adding View model in UI