add more visualization