improved plotting again