[OTHER] Add few comments in the code