Cleaned up visualization.js