Edited some outputs for fig