Added if __name__ ... code to __init__.py