Improve result vizualizator demo