Added ability to connect to DB. Still can't pass the unit test.