Added a couple of comments to make code more coherent