Ruby test-code problem solving!