add db imports to models