Refactoring Mantel test plotting