Updated km.allplot to account for train/test data