Fixed bug in warn also changed to highlight badge