Thanks a lot Jurgen for the help.
I have tested it and it works for the material in change or create mode.
I have one more question :
This change will be for all the materials that use this class and also for all future materials that will use the class. Is this statement correct ?
I have tested it and works but just wanted to confirm with you one more time.