Add Class for extension module