updated the imports on __init__