Added the main function to the header file.