Refactored an if statement...