added comments to plot
maybe fix?