Adding output option to script