add input arg to main