Fixed the base class init.
Added to the __init__.py file.