Nugit's Python Data Scientist Task