Improved plotting example a bit