Final Layer of language model