Modify major mode that which-function-mode enables