Adding badge instead of just a link