Part of Data Scientist toolbox Assignment by Abhinav Saxena