Added test for language model