change badge() method to set an int