modified the UI

Signed-off-by: Senal Kumarage <senal@dotbell.com>