Improve Thrift generation task