Add change in visualization for testing purposes