Added streamlined support for setting output.

So far, this is only used when plotting models.

Signed-off-by: Jakob Brandsborg Ehmsen <jakobehmsen.github@gmail.com>