- Check about using trait id instead of trait name in queries in data_set.py

- Ask Rob about Probe/cellid traits