Finish problem 2 - ruby