Unfinished Language Model implementation