added import for db in models