Add a TDD sample to the testing section
Update the test to work with current code