Add stuff to __init__.py