part of the final project in the Coursera course "Data Scientist's Toolbox"