import db in __init__.py