Update to read db connection