Update visualization code
Merge pull request #4 from ChimeraTeam/codeUpdate

Update visualization code