Added missing __init__ import