add subset.py