Add exports in __init__