Added a softmax policy function. Doesn't work. Lots of horrible debug code.