Solved find digits in python

Signed-off-by: Ionut Radu <ihonius@gmail.com>