visualization module comments and clean up
Updated visualization module comments
Merge pull request #4 from zzhang87/master

visualization module comments and clean up