attempting to fix output figure issues