Should have updated __init__.py in last commit