Change __init__ so that setup.py works