Updated digits from scratch example