Remove unnecessary __init__.py
Diffusion refactor