Improve dataset section