Created for the data science toolkit coursework