Added header files and create main class from class homework.