Added a way to save output