(refactor) Change db connection