Update function doc comment with @return