output modified

I have modified the output of the script according to what you said in the issue