Added classes and unit test for topic modelling