Add header file for proper compilation