initial code for data analysis