Update lda_module.py