add type to __init__