Add functions to create Fig. E2