UI changes

for showing the badge