Modified the array to use comprehention.