I was wondering who wrote those solutions in the yellow paper found by the grandmother near the end of the movie?

I am inclined to think that it was Diane who wrote those solutions, but the drawings and the "YES" made me think otherwise.