I'm very interested in Fig. 6 in your paper. Could you please share the code for it? 