Add comments explaining the code