Simplifying the write_csv() function