added additional imports to init.py