Add unit tests to make the code more robust
Add more code documentation for the Unigram_Model components
Implement fancier models in later versions
Implement extensions to the LDA model in later versions