Clean up errors in distribute upgrade.
Add digits dataset.