Update visualization part