Add convenience imports to __init__