Update __init__.py

adding missing os module import