Fix the plotting to be more consistent.
Fix the issue with the index being returned as floats but only when we don't have any NaN values in the indices (NaNs being returned when the requested position exceeds the distance threshold for the nearest model node).