add unit tests 80% code coverage