Add comments to make code more readable