refactor in only 6 lines. I :heart: ruby