Added two missing header files