# select the index of a single datapoint - the first one in the dataset
idx = 0