Adds testing for badge feature