Solved problem 8 in ruby