added a class attribute to the existing diagnostics