For Data Science Toolkit course project