Add testing coverage