Updated __init__