added code for figures for ICASSP paper