Fixed bug in visualisation
Merge pull request #53 from fvanderlinden/master

Fixed bug in visualisation