Minor change in argument check

Improved readability. Functionality the same
Improved code

Changed the first if expression.
Made the code more readable, the functionality is the same