Adding in DB connectivity