Added code coverage functionality