add setBadge method
add a little bit of documentation about the class