new additions/bug fixes for ED2 plotting