Make the visualization a bit nicer