Made changes according to ui.c.

Reviewer : @poolakiran