Restructuring
Update __init__.py