Changed __init__ to import form the right place