Update __init__.py
Merge pull request #2 from BowenXiao/patch-3

Update __init__.py