Update codewars.rb