adding this to work with db