code for generating figures in my Low-rank Optimal LDA