using autoclass make more sense then automodule

* This is because all this classes inherit from the base class