Move soft_wraps in __init__.py