Added visualization code (paint by the numbers)