Added the checkpoint directory to gitignore. Added the changes requested by Ethan, namely plotting the vertical components of the diffusion term to be more consistent.