add db logic