added import statement to __init__