Update comment for function