Update __init__.py

Added provisions for dimensions with the rates call.