Clean __init__.py
Clean code