clean up and some plotting of probabilities