Adding db connect back