Updated stubbing and unit tests for DB class

More stuff for the DB class