DOC: improving esxamples in small toy
TEST: correcting doctests