tweak plotting
bug fix