added files
added 3 lines in main