Added visualization part