Add basic comments to the code