Added visualization functions