Add __init__ to module