Added test  coverage