Add new source file for main function