FIX: Edited __init__.py to make sure classes are imported
FIX: Edited __init__.py to make sure classes are imported