Adding coverage to test