Added data collection script