Improve dataset upload flow