adding some unit testing for coverage