Allow to add a custom class to notification