Moving all classes to __init__.py file