Minor problem in command-line argument check fixed