fixed a bug in visualization
Merge pull request #50 from jani-bevk/visualization

Fixed a bug in visualization.