Adding the db connection
fixing issue