updated header files in main