Added #include for missing header file