Working on output for fig 1