refactored style for the visualization and testing section