Add method to get current badge number