reworked neuralnets_digits notebook