Re-hooked Expectimax with new API