Add description of code in README, as well as headers for the images.