Fixed some formatting in __init__.py