Adding first integration test (TDD)