Did the original paper show that the toy model was fully grokked?