Refactorisation on test cases. Added more failure probability tests