Updated the input argument and return type.