better figures for paper Sec 3.1