Add code for fig6.