Added tests - code coverage