PY: Small fix and added the data sets that Dave wanted.