[Easy Task] Add comments to existing code for better understanding

### Description of the idea

Add clear, helpful comments to the code to improve readability for beginners. Focus on explaining logic, functions, loops, and conditionals.

✅ Tasks:

- Review the code

- Add meaningful comments (avoid obvious ones)

- Ensure code runs after changes

### Add several screenshots or links related to the idea (if applicable)

_No response_

### Add more details about the idea

_No response_