work in progress -- encode the toy model with the correct normalization