#ifndef CLASS_PDEHEAT
#define CLASS_PDEHEAT

#include "PDE.h"

class PDEHeat : public PDE {
 public:
  PDEHeat(int N, double dt, double a, double b, double theta, double D);
  void setOperator();
  void setBC();
  void setIC();
  
 private:
  double D;
};

#endif