Add __init__ for importing as package