add test coverage
work on test