Don't use a an array to store max probability