Array problem
Minor tweak