Modify thecode to set the buffer size