added code for array comprehension