clean-up and some docstrings
Merge pull request #3 from citenet/visualization

just some sugar..