Ruby solution for problem 4