add topic modelling functions