First question completed
Code for data analysis