Better solution
Allow one argument in command-line