[TASK] Add code comments