Added DB and testing for DB