adding missing imports in init.py