Refactor init.py