Add __init__ again