improved plotting