nearly done with 5. Still need to verify perplexity. Also, there may be a slight bug in the forward backward, since while the probs are good, they don't match the pdf exactly