Minor improvements to natural language inference section