Add comments to the code (better late then never)