Included headers for all new files

Included header files in main for all three new files needed for this
lab.