very minor edit (using "return" keyword for readability)