minor fix
Base class added