Completed milestone 1 with TDD