Code creating figure 7 in the writeup