Adding code for data analysis