Visualization complete. Added main() for user interaction.