Change probmods tests to remove multiviz, add display