Add DB connection and tests