add two arg to the main function