finished TDD assignment