Adding mate tests
Added pruning