code/figure4.m