First attempt at TDD