working on data analysis code