Updating comments in some of the functions