Adding class generator