Printing better for python script