Update the code in the function basis lecture.

I now think that there should be a better example for the Ritz method.