Started on the Crank Nicolson scheme. Seems to be something wrong with the matrix sizes.