Added visualization work.