A version of the initial 2D code with lots of comments. Also, at the top is a few paragraphs describing at a more general level what we are doing in each step of the code