Updated output file for submission :)