Change success messages to class success