--- a/src/include/vtkplot.hpp
+++ b/src/include/vtkplot.hpp
@@ -5,43 +5,40 @@
 #include <string>
 
 #include <vtkSmartPointer.h>
-#include <vtkPolyData.h>
 
-#include "linalg.hpp"
+#include "interpolator.hpp"
 namespace kwantix{
 
-  ///A class for creating VTK plots
-  class vtkplot{
+///A class for creating VTK plots
+template<typename RBF>
+class vtkplot{
+public:
+  /*! \brief Constructor performs a Delaunay triangulation on data
+   * \param data - An \f$n \times 3\f$ matrix where the first two
+   *               columns are points in the x-y plane on which a
+   *               Delaunay triangulation will be done, and the
+   *               third column is the value at this point.
+   */
+  vtkplot(const interpolator<RBF>& data);
+  ///Defines the view direction when plotting.
+  void set_view_direction(const vector& dir);
+  ///Without changing the triangulation, update the values.
+  void update_values(const vector& values);
+  ///Enable offscreen plotting.
+  void plot_offscreen(bool yesno);
+  ///Set the base filename used for saving, to which numbers are appended.
+  void set_base_filename(const std::string& basename);
+  ///Actually do the plot, whether offscreen or onscreen.
+  void plot() const;
 
-  public:
-    /*! \brief Constructor performs a Delaunay triangulation on data
-     * \param data - An \f$n \times 3\f$ matrix where the first two
-     *               columns are points in the x-y plane on which a
-     *               Delaunay triangulation will be done, and the
-     *               third column is the value at this point.
-     */
-    vtkplot(const matrix& data);
-    ///Defines the view direction when plotting.
-    void set_view_direction(const vector& dir);
-    ///Without changing the triangulation, update the values.
-    void update_values(const vector& values);
-    ///Enable offscreen plotting.
-    void plot_offscreen(bool yesno);
-    ///Set the base filename used for saving, to which numbers are appended.
-    void set_base_filename(const std::string& basename);
-    ///Actually do the plot, whether offscreen or onscreen.
-    void plot() const;
-
-  private:
-    ///Base filename
-    string basename;
-    ///Direction from which to plot
-    vector view_direction;
-   ///The plot data
-    vtkSmartPointer<vtkPolyData> polydata;
-    ///Whether to plot offscreen or not
-    bool offscreen;
-  };
+private:
+  ///Base filename
+  string basename;
+  ///Direction from which to plot
+  vector view_direction;
+  ///Whether to plot offscreen or not
+  bool offscreen;
+};
 
 }