about to remove component-related code form dataset.py