This is part of the Coursera "Data Scientist's Toolbox" course project