Changed choose_action to return whole probability table for actions