Add BadgeNotifications class - This is just a mock at the moment