Add digit dataset and digit python