This is a part of data science course project