More idiomatic Ruby please