Update __init__.py

Minor changes to track what was done in master