[Close #3] Add saved output + display refactor