made visualization a bit nicer