modify db class

modify db class