Fixing the outputs of the two files. Nasty hacky code :)