Updated visualization.
Merge pull request #38 from jzvelc/master

Updated visualization.