How to format the output in Python