Returning values from .writerow()