Slight modification to make setting up output file more intuitive