Forgot to add changes to __init__.py in last commit. Adding.