Small comments improvement