how to store output to file in python