Adding some comments

Skimming through at the moment just to try get a sense of the structure
of the program. Will add more later.