Moved the include into the header file