Move init to __init__