Added topic model code