added support for setting badges