Cleaned up visualization example for sharing