Tdd task with balls and boxes