Put back the whitespace, and update the function