Add ability to set a badge, close #24