Add non-const overload of operator()