Adding test set for digits