Isn't this code more complicated than the `.each` that was there initially?