Optimize language model to only include keywords