Edit the function