Add badge to show count