[DOC] A few comments to help refactor the code