Tidying code
Mock render probability mass as PieChart