Some code as part of chapter21 grokking