Adding LDA support