Updated the value function and policy visualization classes to use the
new 2D RenderLayer pipeline.