Updating visuals code