include a few more modules in __init__.py