Saving Files In Python Using Return