Added hooks to display in addition to save plots