Support for while in array comprehension