Adding some comments to the code, nothing big