For the Data Scientist\'s Toolkit class project