Update __init__.py

#2 fixing import statement. Moving it up one folder.