can use command line argument
small modify