Improve badge styles