correctly set Y axis max for probabilities
optimisation: no need to work on a temp copy
pep-8