Add test with coverage