Remove hardcode in getting DB connection