Added some badge functionality.