Add ability to plot either input or output (reducing amount of default output)