Add support for command-line arguments
Only use command-line arguments if two are given.