Fixed some probability bugs and pretty prints