Update in comments as per the previous commit