add db code

add db code