Add tests for command-line arguments