Assignment task - Data Scientist's Toolbox