Added badge class to notifications