Fix buggy training code. Check for special word *
Add tests based off of the corpus.