You can also use "Variable highlights" and use one of the procedural functions, or get the exact same result as "thd777" suggests with "Edit Function", forces you into the function editor :)