Optimize comments

[DavideC03]

Is your feature request related to a problem? Please describe.

At the moment the code is not properly commented

Describe the solution you'd like

Fix comments

Issue Author

@DavideC03