Added more topic model code