Removed if statement in favor of a one line solution.