[FIX] fixes to plotting functions