Update __init__.py

changed extend