add db class for test