Slight refactor of the answers for Array