Refactor to return stack_potentialresponse_tree_state not array.

This should make the code a bit easier to work with.

Also clean up some whitespace.