Added test cases for language model