added NSArray solution?
update