A simple lda implementation in python