Day 6 coding, TDD again