Added cleanup code to clear visualization when .... is chosen