visualization module comments and clean up