Create subset.py