More for CNNs. Added test with digits dataset