Small change in __init__