Minor changes. VTK into function to speed up loading.