Add <functional> to files that use std::function