# Set problem parameters
I = 1.2
a = 0.2
T = 8
dt = 0.25
theta = 0.5
from solver import solver, exact_solution
u, t = solver(I, a, T, dt, theta)
import matplotlib.pyplot as plt
plt.plot(t, u, t, exact_solution)
plt.legend(['numerical', 'exact'])
plt.show()