[DOCS] Add more comments to the code

[Maghish]

Describe the problem/idea:
Since this project is open-source, add more comments.

Your solution: (if there is any)

Checks:

• This is very important to fix/add this
• This is an suggestion, not a bug report
• This is a bug report, not a suggestion