Visualization now working!