A new version with animation. A bit complicated with FuncAnimation me thinks...