Updated lda.py