my code in learn tdd in ruby