Minor fixes and updates in build dataset module.