adding TDD to Unit2