refactored the visualization code
fixed untracked files