Modify the function to achieve some