improved output plot