Update simple_LDA.py