Update __init__.py

small formatting fix