created for assignment of data science course 1