Fix bug in language model
Fix bug
Bug shifting with UNK