Created for data scientist toolbox course project