Fix return of Array#each and style of test