Updated Main File

Added some methods to the main file and added some other header and cpp
file that will be needed.