Fixed up visulizer class.