added vtk code for visualization