Update lda.py