First try at Bellman class version. Will use W_guess as a 2D matrix now instead of a vector