Add include for std::function