Add badge to classes