Add __init__ to import as package