updated the util code to buffer mode write

[bhanu77prakash]

As mentioned in the latest issue, I have tried to convert the write mode from line by line to a buffer mode write. I tried running the updated script on the toy dataset and the outputs are exactly same.

[AdrianKs]

Thanks, this is a good idea. Some small changes are needed though.