For Data Science Assignment