Added visualization JavaScript