Added support for header files as well