Added badge for count