Update visualization.lua