For final project of Coursera Data Scientist Toolbox course