Update __init__.py

Fixed documentation to match code argument order.