Fixed visualization
Merge pull request #31 from as11712/master

Fixed visualization