Add textbox to visualization, smoother animation
Refactor text display for clarity