Added styling for badge