Test driven development part 2