UI changes

Signed-off-by: Dileep Mishra <mishradileep@gmail.com>