Fix up a few bugs in the bellman basis.
Adapt cartpole experiment to new bellman basis.