small change on __init__