Add action to remove visualization