Added a "badge" option to enable/disable the badge feature.