Update comments in final version of the 2 functions